package cn.harry.wx.service.impl;

import cn.harry.config.TaskExcutor;
import cn.harry.wx.entity.WxTemplateMsgLog;
import cn.harry.wx.entity.WxUser;
import cn.harry.wx.form.TemplateMsgBatchForm;
import cn.harry.wx.service.TemplateMsgService;
import cn.harry.wx.service.WxMsgTemplateService;
import cn.harry.wx.service.WxTemplateMsgLogService;
import cn.harry.wx.service.WxUserService;
import com.baomidou.mybatisplus.core.metadata.IPage;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.common.error.WxErrorException;
import me.chanjar.weixin.mp.api.WxMpService;
import me.chanjar.weixin.mp.bean.template.WxMpTemplateMessage;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;

/**
 * @author Nifury
 * @date 2017-9-27
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class TemplateMsgServiceImpl implements TemplateMsgService {
    @Autowired
    private WxTemplateMsgLogService templateMsgLogService;
    private final WxMpService wxService;
    @Autowired
    WxMsgTemplateService msgTemplateService;
    @Autowired
    WxUserService wxUserService;

    /**
     * 发送微信模版消息,使用固定线程的线程池
     */
    @Override
    @Async
    public void sendTemplateMsg(WxMpTemplateMessage msg, String appid) {
        TaskExcutor.submit(() -> {
            String result;
            try {
                wxService.switchover(appid);
                result = wxService.getTemplateMsgService().sendTemplateMsg(msg);
            } catch (WxErrorException e) {
                result = e.getMessage();
            }

            //保存发送日志
            WxTemplateMsgLog log = new WxTemplateMsgLog(msg, appid, result);
            templateMsgLogService.addLog(log);
        });
    }

    @Override
    @Async
    public void sendMsgBatch(TemplateMsgBatchForm form, String appid) {
        log.info("批量发送模板消息任务开始,参数：{}", form.toString());
        wxService.switchover(appid);
        WxMpTemplateMessage.WxMpTemplateMessageBuilder builder = WxMpTemplateMessage.builder()
                .templateId(form.getTemplateId())
                .url(form.getUrl())
                .miniProgram(form.getMiniprogram())
                .data(form.getData());
        Map<String, Object> filterParams = form.getWxUserFilterParams();
        if (filterParams == null) {
            filterParams = new HashMap<>(8);
        }
        long currentPage = 1L, totalPages = Long.MAX_VALUE;
        filterParams.put("limit", "500");
        while (currentPage <= totalPages) {
            filterParams.put("page", String.valueOf(currentPage));
            //按条件查询用户
            IPage<WxUser> wxUsers = wxUserService.queryPage(filterParams);
            log.info("批量发送模板消息任务,使用查询条件，处理第{}页，总用户数：{}", currentPage, wxUsers.getTotal());
            wxUsers.getRecords().forEach(user -> {
                WxMpTemplateMessage msg = builder.toUser(user.getOpenid()).build();
                this.sendTemplateMsg(msg, appid);
            });
            currentPage = wxUsers.getCurrent() + 1L;
            totalPages = wxUsers.getPages();
        }
        log.info("批量发送模板消息任务结束");
    }

}
